package com.erli.exchange.common.sg;

import com.erli.exchange.common.sg.config.BaseClassInfo;

/**
 * Mapper类生成器
 * 
 * @title MapperSourceGenerator.java
 * @package com.example.demo.sourcegenerate
 * @author lihu
 * @update 2019年3月8日 下午3:12:18
 */
public class MapperSourceGenerator extends AbstractSourceGenerator {

	@Override
	protected String getFileMainContent() {
		importSet.add(getFullClassImportByBaseClassInfo(entityClassInfo));
		return "";
	}

	@Override
	protected String getFileFirstLineContent() {
		return String.format(templateInfo.getFileStart(), baseInfo.getInterfaceType(), getClassName(),
				String.format(mapperClassInfo.getExtendsTemplate(), this.getEntityClassName()));
	}

	@Override
	protected BaseClassInfo getSelfClassInfo() {
		return mapperClassInfo;
	}

}
